import tensorflow as tf


# 检查 GPU 是否可用
if tf.config.list_physical_devices('GPU'):
    print("GPU 可用")
else:
    print("GPU 不可用")
